package 每日一题;

import java.util.*;

/**
 * @description:
 * @author: 小白白
 * @create: 2021-12-16
 **/

public class No851喧闹和富有 {

    /**
     * 有一组 n 个人作为实验对象，从 0 到 n - 1 编号，其中每个人都有不同数目的钱，以及不同程度的安静值（quietness）。为了方便起见，我们将编号为 x 的人简称为 "person x "。
     * 给你一个数组 richer ，其中 richer[i] = [ai, bi] 表示 person ai 比 person bi 更有钱。另给你一个整数数组 quiet ，其中 quiet[i] 是 person i 的安静值。richer 中所给出的数据 逻辑自洽（也就是说，在 person x 比 person y 更有钱的同时，不会出现 person y 比 person x 更有钱的情况 ）。
     * 现在，返回一个整数数组 answer 作为答案，其中 answer[x] = y 的前提是，在所有拥有的钱肯定不少于 person x 的人中，person y 是最安静的人（也就是安静值 quiet[y] 最小的人）。
     * <p>
     * 示例 1：
     * 输入：richer = [[1,0],[2,1],[3,1],[3,7],[4,3],[5,3],[6,3]], quiet = [3,2,5,4,6,1,7,0]
     * 输出：[5,5,2,5,4,5,6,7]
     * 解释：
     * answer[0] = 5，
     * person 5 比 person 3 有更多的钱，person 3 比 person 1 有更多的钱，person 1 比 person 0 有更多的钱。
     * 唯一较为安静（有较低的安静值 quiet[x]）的人是 person 7，
     * 但是目前还不清楚他是否比 person 0 更有钱。
     * answer[7] = 7，
     * 在所有拥有的钱肯定不少于 person 7 的人中（这可能包括 person 3，4，5，6 以及 7），
     * 最安静（有较低安静值 quiet[x]）的人是 person 7。
     * 其他的答案也可以用类似的推理来解释。
     * 示例 2：
     * 输入：richer = [], quiet = [0]
     * 输出：[0]
     *  
     * 提示：
     * n == quiet.length
     * 1 <= n <= 500
     * 0 <= quiet[i] < n
     * quiet 的所有值 互不相同
     * 0 <= richer.length <= n * (n - 1) / 2
     * 0 <= ai, bi < n
     * ai != bi
     * richer 中的所有数对 互不相同
     * 对 richer 的观察在逻辑上是一致的
     */

    public int[] loudAndRich(int[][] richer, int[] quiet) {

        /**
         * 1. 先得出比 index 有钱的所有集合
         * 2. 在与quiet联合,比对得出最安静的
         */
        // key为没钱人 value为有钱人
        Map<Integer, Set<Integer>> listMap = new HashMap<>(richer.length);

        for (int[] arr : richer) {
            // 有钱人
            int a = arr[0];
            // 没钱人
            int b = arr[1];
            Set<Integer> list = listMap.getOrDefault(b, new HashSet<>());
            list.add(a);
            listMap.put(b, list);
        }

        // 比index有钱的人
        List<PriorityQueue<Integer>> allList = new ArrayList<>();

        for (int i = 0; i < quiet.length; i++) {
            PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(Comparator.comparingInt(i2 -> quiet[i2]));
            allList.add(priorityQueue);
            this.dfs(priorityQueue, listMap, i);
        }

        // 然后开始了
        int[] answers = new int[quiet.length];

        for (int i = 0; i < answers.length; i++) {
            Integer peek = allList.get(i).peek();
            answers[i] = peek == null ? i : quiet[peek] <= quiet[i] ? peek : i;
        }


        return answers;
    }

    private void dfs(PriorityQueue<Integer> priorityQueue, Map<Integer, Set<Integer>> listMap, Integer personIndex) {
        // 比personIndex有钱的人加入队列
        Set<Integer> set = listMap.get(personIndex);
        if (set == null || set.size() == 0) {
            return;
        }
        for (Integer i : set) {
            priorityQueue.add(i);
            this.dfs(priorityQueue, listMap, i);
        }
    }

    public static void main(String[] args) {
        No851喧闹和富有 n = new No851喧闹和富有();
        int[][] arr1 = {{1, 0}, {2, 1}, {3, 1}, {3, 7}, {4, 3}, {5, 3}, {6, 3}};
        int[] arr2 = {3, 2, 5, 4, 6, 1, 7, 0};
        int[] result = n.loudAndRich(arr1, arr2);
        System.out.println(Arrays.toString(result));
    }

}
